package com.zzxz.practicaltrainingsys.dto;

import com.alibaba.fastjson.annotation.JSONField;
import lombok.Setter;

@Setter
public class OllamaRequest {

    private String model;
    private String prompt;
    private boolean stream = false;

    @JSONField(name = "num_ctx")
    private Integer numCtx = 4096;

    @JSONField(name = "temperature")
    private Double temperature = 0.7;

    // Constructors
    public OllamaRequest() {}

    public OllamaRequest(String model, String prompt) {
        this.model = model;
        this.prompt = prompt;
    }

    // Getters and Setters
    public String getModel() { return model; }
    public void setModel(String model) { this.model = model; }

    public String getPrompt() { return prompt; }
    public void setPrompt(String prompt) { this.prompt = prompt; }

    public boolean isStream() { return stream; }
    public void setStream(boolean stream) { this.stream = stream; }

    public Integer getNumCtx() { return numCtx; }
    public void setNumCtx(Integer numCtx) { this.numCtx = numCtx; }

    public Double getTemperature() { return temperature; }
    public void setTemperature(Double temperature) { this.temperature = temperature; }
}